from tortoise import BaseDBAsyncClient


async def upgrade(db: BaseDBAsyncClient) -> str:
    return """
        ALTER TABLE `xm_category` ADD `cate_list` VARCHAR(255)   COMMENT '列表模板';
        ALTER TABLE `xm_category` ADD `cate_show` VARCHAR(255)   COMMENT '详情模板';"""


async def downgrade(db: BaseDBAsyncClient) -> str:
    return """
        ALTER TABLE `xm_category` DROP COLUMN `cate_list`;
        ALTER TABLE `xm_category` DROP COLUMN `cate_show`;"""
